def cola_supplement(stock, *args):
    # 统计每种可乐需要的总数
    required = {key: 0 for key in stock}
    for guest in args:
        for cola in guest:
            required[cola] += 1

    # 计算需要补充的数量
    supplement = {}
    for cola, count in required.items():
        if count > stock[cola]:
            supplement[cola] = count - stock[cola]

    return supplement


# 初始库存
initial_stock = {"可口可乐": 10, "百事可乐": 2, "零度可乐": 0, "芬达": 6, "健怡可乐": 4}

# 每位客人可能喝的可乐品牌
p1 = {"可口可乐", "零度可乐", "芬达", "健怡可乐"}
p2 = {"百事可乐", "零度可乐", "健怡可乐"}
p3 = {"可口可乐", "百事可乐", "芬达"}
p4 = {"零度可乐", "芬达", "健怡可乐"}
p5 = {"可口可乐", "百事可乐", "健怡可乐"}
result = cola_supplement(initial_stock, p1, p2, p3, p4, p5)
print("需要补充的可乐有：", result)